# Create an interactive plot that shows moving average sentiment of state mentions over timeimport plotly.express as pximport numpy as npimport pandas as pd# Load data for mentions over timedata = pd.read_csv('tmp/nlp_results.csv')# Create year week variabledata['date'] = pd.to_datetime(data['date'])data['year_week'] = data['date'].dt.strftime('%Y-%U')# Group data by year week, average sentimentgrouped_dat = data.groupby(['date', 'states_mentioned']).agg(average_sentiment=('sent', 'mean')).reset_index()grouped_dat['moving_average'] = grouped_dat.groupby(['states_mentioned'])['average_sentiment'].transform(lambda x: x.rolling(14, 1).mean())print(grouped_dat.head(10))